package com.alex.megacj.scenario;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.springframework.beans.factory.annotation.Required;

public class ScenarioManager {

	private Scenario defaultScenario;

	private Map<String, Scenario> scenarios = new HashMap<String, Scenario>();

	@Required
	public void setDefaultScenario(Scenario defaultScenario) {
		this.defaultScenario = defaultScenario;
	}

	@Required
	public void setScenarios(List<Scenario> scenarios) {
		for (Scenario scenario : scenarios) {
			this.scenarios.put(scenario.getMirrorHost(), scenario);
		}
	}

	public Scenario getScenario(String mirrorHost, boolean returnDefault) {
		Scenario scenario = scenarios.get(mirrorHost);
		if (scenario != null) {
			return scenario;
		} else if (returnDefault) {
			return defaultScenario;
		} else {
			return null;
		}
	}
}
